package kyssion.leetcode.num201_250;

import java.util.HashMap;

/**
 * 给定一个整数数组和一个整数 k，判断数组中是否存在两个不同的索引 i 和 j，
 * 使得 nums [i] = nums [j]，并且 i 和 j 的差的绝对值最大为 k。
 * <p>
 * 示例 1:
 * <p>
 * 输入: nums = [1,2,3,1], k = 3
 * 输出: true
 * 示例 2:
 * <p>
 * 输入: nums = [1,0,1,1], k = 1
 * 输出: true
 * 示例 3:
 * <p>
 * 输入: nums = [1,2,3,1,2,3], k = 2
 * 输出: false
 */
public class code219_存在重复数字2 {

    public static void main(String[] args) {
        System.out.println(new code219_存在重复数字2().containsNearbyDuplicate(
                new int[]{
                        1,2,3,1
                }, 3
        ));
    }


    /**
     * 直接hash 应为已经有序 而且题目中要求的是最小的数字,所以相同的选择最靠后的就可以了
     * @param nums
     * @param k
     * @return
     */
    public boolean containsNearbyDuplicate2(int[] nums, int k) {
        HashMap<Integer, Integer> map = new HashMap<>();
        for (int i = 0; i < nums.length; i++) {
            if (map.containsKey(nums[i])) {
                int deff = map.get(nums[i]);
                if (i - deff <= k) {
                    return true;
                }
            }
            map.put(nums[i], i);
        }
        return false;
    }


    /**
     * 使用方法形成维度 速度比较慢
     * @param nums
     * @param k
     * @return
     */
    public boolean containsNearbyDuplicate(int[] nums, int k) {
        int[] indexs = new int[nums.length];
        for (int a = 0; a < indexs.length; a++) {
            indexs[a] = a;
        }
        quickSort(nums, indexs, 0, nums.length - 1);

        int start = 1;

        while (start < nums.length) {
            if (nums[start] == nums[start - 1]) {
                if (Math.abs(indexs[start] - indexs[start - 1]) <= k) {
                    return true;
                }
            }
            start++;
        }
        return false;
    }

    public void quickSort(int[] arr, int[] indexs, int start, int end) {
        if (start < end) {
            int midle = partition_s(arr, indexs, start, end);
            quickSort(arr, indexs, start, midle - 1);
            quickSort(arr, indexs, midle + 1, end);
        }
    }

    /**
     * 快速排序获得中间量方法2
     *
     * @param arr
     * @param start
     * @param end
     * @return
     */
    private int partition_s(int[] arr, int[] indexs, int start, int end) {
        int beforeNum = arr[start];
        while (start < end) {
            while (start < end && arr[end] >= beforeNum) {
                end--;
            }
            exchange(arr, start, end);
            exchange(indexs, start, end);
            while (start < end && arr[start] <= beforeNum) {
                start++;
            }
            exchange(arr, start, end);
            exchange(indexs, start, end);
        }
        return start;
    }

    private void exchange(int[] arr, int a, int b) {
        int i = arr[a];
        arr[a] = arr[b];
        arr[b] = i;
    }
}
